package com.gp.pj.aspect;

import cn.hutool.core.util.ClassUtil;
import cn.hutool.core.util.StrUtil;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.datatype.jdk8.Jdk8Module;
import com.fasterxml.jackson.datatype.jsr310.JavaTimeModule;
import com.fasterxml.jackson.module.paramnames.ParameterNamesModule;
import com.gp.pj.annotation.Dict;
import com.gp.pj.service.DictService;
import com.gp.pj.utils.ResultUtils;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.List;

/**
 * @author Chloe
 */
@Aspect
@Component
@Slf4j
public class DictAspect {

    /**
     * 字典项文本后缀
     */
    private final String DICT_TEXT_SUFFIX = "DictText";

    @Autowired
    private DictService dictService;

    // 定义切点Pointcut
    @Pointcut("execution(* com.gp..*.*Controller.*(..))")
    public void excudeService() {
    }

    @Around("excudeService()")
    public Object doAround(ProceedingJoinPoint pjp) throws Throwable {
        long time1 = System.currentTimeMillis();
        Object result = pjp.proceed();
        long time2 = System.currentTimeMillis();
        log.debug("获取JSON数据 耗时：" + (time2 - time1) + "ms");
        long start = System.currentTimeMillis();
        this.parseDictText(result);
        long end = System.currentTimeMillis();
        log.debug("解析注入JSON数据  耗时" + (end - start) + "ms");
        return result;
    }

    /**
     * 本方法针对返回对象为R 的PageUtils的分页列表数据进行动态字典注入
     * 字典注入实现 通过对实体类添加注解@dict 来标识需要的字典内容,字典分为单字典code即可 ，table字典 code table text配合使用与原来jeecg的用法相同
     * 示例为SysUser   字段为sex 添加了注解@Dict(dicCode = "sex") 会在字典服务立马查出来对应的text 然后在请求list的时候将这个字典text，已字段名称加_dictText形式返回到前端
     * 例输入当前返回值的就会多出一个sex_dictText字段
     * {
     * sex:1,
     * sex_dictText:"男"
     * }
     * 前端直接取值sext_dictText在table里面无需再进行前端的字典转换了
     * customRender:function (text) {
     * if(text==1){
     * return "男";
     * }else if(text==2){
     * return "女";
     * }else{
     * return text;
     * }
     * }
     * 目前vue是这么进行字典渲染到table上的多了就很麻烦了 这个直接在服务端渲染完成前端可以直接用
     */
    private void parseDictText(Object result) {
        if (result instanceof ResultUtils) {
            // 判断Result的result类型是不是IPage
            if (((ResultUtils) result).getData() instanceof IPage) {
                List<JSONObject> items = new ArrayList<>();
                // 遍历这些分页记录
                for (Object record : ((IPage) ((ResultUtils) result).getData()).getRecords()) {
                    ObjectMapper mapper = new ObjectMapper();
                    String json = "{}";
                    try {
                        // 解决 LocalDateTime 在 mapper.writeValueAsString(record) 转换失败的问题
                        mapper.registerModule(new ParameterNamesModule())
                                .registerModule(new Jdk8Module())
                                .registerModule(new JavaTimeModule());
                        // 解决@JsonFormat注解解析不了的问题详见SysAnnouncement类的@JsonFormat
                        // 将序列化值转为String？？（另一种意思，由于已经是分页记录了，因此获取分页记录中的每个，然后将其转化为JSon格式的String，再然后，转成JSonObject）
                        json = mapper.writeValueAsString(record);
                    } catch (JsonProcessingException e) {
                        log.error("json解析失败" + e.getMessage(), e);
                    }
                    // 将json数据转化为jsonObject类型。
                    JSONObject item = JSONObject.parseObject(json);
                    // for：解决继承实体字段无法翻译问题------
                    // 获得这些分页记录中的变量
                    for (Field field : ClassUtil.getDeclaredFields(record.getClass())) {
                        // 判断该变量上是否有@Dict注解
                        if (field.getAnnotation(Dict.class) != null) {
                            String code = field.getAnnotation(Dict.class).dicCode();
                            String text = field.getAnnotation(Dict.class).dicText();
                            String table = field.getAnnotation(Dict.class).dictTable();
                            // 获得之前jsonObject中的该变量名对应的值。
                            String key = String.valueOf(item.get(field.getName()));

                            // 翻译字典值对应的txt
                            String textValue = translateDictValue(code, text, table, key);

                            log.debug(" 字典Val : " + textValue);
                            log.debug(" __翻译字典字段__ " + field.getName() + DICT_TEXT_SUFFIX + "： " + textValue);
                            item.put(field.getName() + DICT_TEXT_SUFFIX, textValue);
                        }
                    }
                    items.add(item);
                }
                // 最后把已经转换好了给换了。
                ((IPage) ((ResultUtils) result).getData()).setRecords(items);
            }
        }
    }

    /**
     * 翻译字典文本
     *
     * @param code  字典的编码，如性别的编码是sex。
     * @param text  要查询的表中指定字段，如，需要查询dict_name，或description什么的
     * @param table 数据字典表，不为空时，指定到该字典表进行查询。
     * @param key   字典项值，如，sex 为1 表示男，这个 1 就是字典项值，结合code（此例子中code应为sex）使用。该值对应的才是真正需要的数据。
     * @return key对应的值
     */
    private String translateDictValue(String code, String text, String table, String key) {
        if (StrUtil.isBlank(key)) {
            return null;
        }
        // 之前用的是StringBuffer，虽然是线程安全的，但没必要，效率还低一点。
        StringBuilder textValue = new StringBuilder();
        String[] keys = key.split(",");
        for (String k : keys) {
            String tmpValue;
            log.debug(" 字典 key : " + k);
            k = k.trim();
            if (k.length() == 0) {
                continue; //跳过循环
            }
            if (!StrUtil.isEmpty(table)) {
                tmpValue = dictService.queryTableDictTextByKey(table, text, code, k);
            } else {
                tmpValue = dictService.queryDictTextByKey(code, k);
            }

            if (tmpValue != null) {
                if (!"".equals(textValue.toString())) {
                    textValue.append(",");
                }
                textValue.append(tmpValue);
            }

        }
        return textValue.toString();
    }
}
